from PIL import Image
import os

# 图片素材的文件夹路径
input_folder = 'D:\CIFAR10\pytorch-cifar\input_images'  # 输入图片素材文件夹路径
output_folder = 'D:\CIFAR10\pytorch-cifar\output_images'  # 输出图片保存文件夹路径

# 创建输出文件夹，如果不存在的话
if not os.path.exists(output_folder):
    os.makedirs(output_folder)

# 遍历文件夹中的所有图片
for file_name in os.listdir(input_folder):
    if file_name.lower().endswith(('.png', '.jpg', '.jpeg')):  # 只处理图片格式
        # 打开图片
        img_path = os.path.join(input_folder, file_name)
        img = Image.open(img_path)

        # 获取原始尺寸
        original_width, original_height = img.size

        # 转换图片为PNG格式，并调整大小为32x32
        img_resized = img.resize((32, 32))  # 调整为96x96像素
        output_path = os.path.join(output_folder, os.path.splitext(file_name)[0] + '.png')  # 输出为PNG格式

        # 保存图片为PNG
        img_resized.save(output_path, format='PNG')

        # 输出处理信息（包含原始尺寸）
        print(f'已处理图片: {file_name}（原始尺寸：{original_width}x{original_height}） -> {output_path}')

print("所有图片处理完成！")